Artificial intelligence (AI)-based predictive model